#ifndef _VIDEOFILE_H_
#define _VIDEOFILE_H_

#include "opencv2/opencv.hpp"
#include "preprocess.h"

#include <string>

class VideoFile
{
public:
    VideoFile(const std::string &&filename);
    ~VideoFile();
    void Display(const float framerate = 25.0, const int target_size = 640);
    std::unique_ptr<cv::Mat> GetNextFrame();
    cv::Mat test();
    int get_frame_width();
    int get_frame_height();

private:
    std::string filename_;
    cv::VideoCapture *capture_{nullptr};
};

#endif // _VIDEOFILE_H_